package cn.tedu.boot02.controller;

import cn.tedu.boot02.entity.User;
import cn.tedu.boot02.util.DBUtil;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

@RestController
public class UserController {
    @RequestMapping("/reg")
    public String reg(User user){
        System.out.println("user = " + user);
        try( Connection conn= DBUtil.getConnection()) {
           String sql = "SELECT id FROM user WHERE username=?";
            PreparedStatement ps= conn.prepareStatement(sql);
            ps.setString(1, user.getUsername());
            ResultSet rs= ps.executeQuery();
            if(rs.next()){
                return "用户名已存在!";
            }
            //代表用户名可用
            String insertSQL = "INSERT INTO user VALUES(NULL,?,?,?)";
            PreparedStatement insertPS = conn.prepareStatement(insertSQL);
            insertPS.setString(1, user.getUsername());
            insertPS.setString(2, user.getPassword());
            insertPS.setString(3, user.getNickname());
            insertPS.executeUpdate();


        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }


        return "注册成功！";
    }
    @RequestMapping("/login")
    public String login(User user){
        System.out.println("user = " + user);
        try(Connection conn=DBUtil.getConnection()) {
            String sql = "SELECT password FROM user WHERE username=?";
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1, user.getUsername());
            ResultSet rs =ps.executeQuery();
            if(rs.next()){
                //执行到这里代表查询到了

                String pw=rs.getString(1);
                //判断用户输入的密码和数据库里查询到的是否一致
                if(pw.equals(user.getPassword())){
                    return "登录成功！";
                }
                return "密码错误！";
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return "用户名不存在！";
    }

}
